#pragma once
#include "Common.h"
//跨平台的方法:条件编译
// #ifdef _WIN32
// 	#include<windows.h>
// #else
//     #include <unistd.h>
//     #include <stdio.h>
//     void* custom_alloc(size_t size) {
//     void *curr_brk = sbrk(0);  // 获取当前program break地址[2,3](@ref)
//     if (brk(curr_brk + size) == -1) {  // 尝试扩展堆空间
//         perror("brk failed");
//         return NULL;
//     }
//     return curr_brk;  // 返回分配前的地址作为可用内存首地址[2,7](@ref)
// }
// #endif

// // 直接去堆上按页申请空间
// inline static void* SystemAlloc(size_t kpage)
// {
// #ifdef _WIN32
// 	void* ptr = VirtualAlloc(0, kpage<<13, MEM_COMMIT | MEM_RESERVE, PAGE_READWRITE);
// #else
// 	// linux下brk mmap等
//     void *ptr = custom_alloc(kpage);  // 分配4KB内存
// #endif

// 	if (ptr == nullptr)
// 		throw std::bad_alloc();

// 	return ptr;
// }

template<class T>
class ObjectPool
{
public:
	T* New()
	{
		T* obj = nullptr;

		// 优先把还回来内存块对象，再次重复利用
		if (_freeList)
		{
			void* next = *((void**)_freeList);
			obj = (T*)_freeList;
			_freeList = next;
		}
		else
		{
			// 剩余内存不够一个对象大小时，则重新开大块空间
			if (_remainBytes < sizeof(T))
			{
				_remainBytes = 128 * 1024; //一次开了128*1024KB,如果一页为8KB,这里就是开了16页

				//_memory = (char*)malloc(_remainBytes);
				_memory = (char*)SystemAlloc(_remainBytes >> 13);
				if (_memory == nullptr)
				{
					throw std::bad_alloc();
				}
			}

			obj = (T*)_memory;
			size_t objSize = sizeof(T) < sizeof(void*) ? sizeof(void*) : sizeof(T);
			_memory += objSize;
			_remainBytes -= objSize;
		}

		// 定位new，显示调用T的构造函数初始化
		new(obj)T;

		return obj;
	}

	void Delete(T* obj)
	{
		// 显示调用析构函数清理对象
		obj->~T();

		// 头插
		*(void**)obj = _freeList;
		_freeList = obj;
	}

private:
	char* _memory = nullptr; // 指向大块内存的指针
	size_t _remainBytes = 0; // 大块内存在切分过程中剩余字节数

	void* _freeList = nullptr; // 还回来过程中链接的自由链表的头指针
};

struct TreeNode
{
	int _val;
	TreeNode* _left;
	TreeNode* _right;

	TreeNode()
		:_val(0)
		, _left(nullptr)
		, _right(nullptr)
	{}
};